Adding GPU availability check