Added tests for assignment