Update to flat design