improved layout + restructured code + inserted plots