Helo Anila
You as the architect can change your domain model to reflect a solution designed to solve all use cases. Do not forget to document the reasons.
Remember, the assignament may be missing a lot of information in which you will have to assume in order to develop and deliver your proposed solution.
Regards.