Turned code in Ross' notebook into python code. Still need to implement change_one_pixel function.