Assignment attempt 3 - With comments