first part of HW1. Not everything is done yet