Added testing lines on class