Updated the shaders and some clean up in the code