adding the code so far