Lab6 - 2 DONE (no red lines / not sure it's working)