Updated GetModule implementation and added more tests