Add support for wheels