UI changes

Going for a flatter style