Wrote initial code to read in input