added basic messages and fixed plot