Added functions to generate plots used in thesis