Codejam python engine for weather prediction