updated README, cleaned code