cleaned code
clean up code