cleaned code and readme