Add first_epoch_step_monitor to MNIST experiment