Added tests for storing and loading registers