Assignment 4 (has problem)