updated readme
Iteration 2. Some additional logic and pep8 compliance