Update socket
Merge pull request #2 from shreyash07/socket-edits

Update socket