restored lines i removed from main module