Removed debug println
Added tests