Edit README and modify code to utilize list comprehensions
Edit README and modify code to utilize list comprehensions