Day 2 code (pre class)