Added initial version of styling to the UI