working on WORK_FAIL
added class diagram