Minor UI fixes

fixed issues when switching between heat map and gantt