clean up

cleaned up code