Modified coding to place wheels check within __init__