need to fix graph axes and do some parameter tuning