Added test cases for testing logic with Storage