Updating TODO
Adding back Math.max logi