Tried to clean code up. Can't remember what exactly I was doing.