Modified state diagram. Added more message arrows