cleaned code, updated readme