fixed learning
added data manipulation and vizualisation