refactored feature plotting