Refactor code to read/write different memory types