fixed something from hw3
fix