corrected assignment, added new test code