Changed python models and wrote migration