code cleanup, comment added