Added communication code