Version for HW4 submission