Finishing session 01 code