Almost finished assignment