[Completion] Module statement completion improvements