Set point size in vertex shader.
Code tidy up.