Added commenting and improved figures