Add code from second module