Adding boiler plate to test classes