probably fixed hw3 bug