Added test to previous assignment
Added test to previous assignment