Add support for drain