Changes a test case sligthly.
Updates class diagram.