Simplify the logic of memory interface
Add small testing graphs