Updated processing and plotting for assignment 1