Move the communication code to a separate file