Added check for wheel in setup