Fix output
Improve based matlab functionality
Improve based matlab functionality (SOR)