Fixed something that someone pointed out
Work done in class