Re-add main lift code