Added missing class to complete build