added the initial code