added visualize.py
seperate features
draw plot