Optimized next_time_step. Only loop over neighbouring atoms once for every atom