Added option to return to main menu and take another picture

My first pull request! Hopefully I didn't break anything.