I have already updated these lines as your comment. Please review them on updated source code. Thank you!