Add support for walls