Got code working, but my plot looks weird.