removed some code

changing some code