fixed code for comments