Add code for taking photo