Final touches

- Some styling here and there