Add license header to py file