add some code for leetcode online judge