[DM HW3] modifications to sol