Fixed #28. Added dismiss instead of back while taking picture and fixed confusion on going back when retraining.