added custom graph and make this work with travis