Added result plotting