Clean Plot grid code and add to documentation