Added required figures and a main function generating all plots at once.