Revert "Revert "Made the shading happen in the fragement shader""

This reverts commit ce49787c575478a5ee9e3448154a0586575adddb.