OpenAI Gym - save as mp4 and display when finished