added comparison plots code