Fix bugs; add new messages