Add flash socket stuff