Update docs and improve styling