made formatting changes to make pylint happy
Added solution to problem 3