Code works but looks like crap, this was an assignment.....