Improvement to install feature