Adding test framework