changing some code up