Adding the code so far