Add prompt support