Made changes to feature module