Implemented option 1 for Lab 0 Q1

Run it together with input_0_1.txt