Wrote initial code