Adding test function