Correct socket imports
Add extra dependencies