Add test running to build