Assignment 2 - start implementation