added tests to models