UI fix: Add back button after photo taken