fine tuning maingraph and getting more junit cases in