shader refactoring
Added comments to fs