Added tests to models