Removed print statements in get_features