Added some code on python