cleaned code

cleaned code