Limiting the axis plot for better resolution
New vizualiation function
Passing the environment as param to learn
Learning until terminal state