added new code
if __name__ stuff added to feature.py