hw_4 almost done