After stage 1

Code I have after finishing stage 1