Add prompt tests