Made various code and image updates post-critique