Cleaned some code up