Now working with socket io