Adding test for methods