Added python header and some comments