Add support for restricting hosts