Added testing code
Added testing code