Updated README and made plots prettier