cleaned code a little bit