updated code comment
fixed test validation