added some comments
added instructions to README
moved input processing to list comprehension
changes