Add functionality to the markup

All tests green, nice. Now add style