Moved some logic to the models from the controllers
Added some css code to show arrow icons