solution for test assignment