This is pretty cool pythonic code for initial contribution :)