Add support for custom phases