Made changes to unify input for each part to match the other modules