Added single unit test