Wrote unit tests for storage