Fixed minor bug and made improvement to memory driver!