modify leetcode 1-5