Added mock framework