Add code behind for step 5