fix tiny bug in time_grids and made plots even prettier