Needs to be cleaned up but works