Updated after running Travis