Add forgotten classes