Improve shader module