Add custom port support