Continuing to implement Euler example
Changed shape of U
Improved code readability