Fixed a label and changed the altered the setup to equation 3