Improved control buttons layout

+ Added buttons for undo/redo;
+ Other styling fixes.