Улучшен код.