bug fixes for hw7