Missed adding these two new classes