cleaned code
finished project