Added the code from first lab