Add step for getting the code