Some refactoring. Moved updating uniform to shader class.