Added tests for inherited classes