More UI changes + Photo input works correctly