Add unit test for Account model