Added testing for comments