Add unit tests for User model