Add cache handling