removed comment
Added test class