Add basic socket support