Added add_custom_feature function
pep8
Add and get custom feature functions added
Removed print statement