Got union type completion working

and all tests passing