Add completed project 2 code