Adding test codes