Converted python notebook code to script so i don't need notebook running to run experiments