Finished Problem 12

Removed endless loop of doom, finished up code that actually works.
Learned a bit about lambda funtion in Python.