Added shader setup code