Added ability to toggle shader stages