Added all the code from the video