Add wrapper classes