Added test for assignment