Fixed docstring and TODOs for after_reply_message_received