Updating to make work better with the endstops