Add stub classes for functionality