updated README, slight modifications to code