Added more code for Leet code