Add actual class I just wrote