Add my code for the first two lessons