Add more stub code