Added testing to test model relationship