Add logic to gantt