Re-work input checks against number of nodes available