need more work for ex4