Completing code addition for stage 1