Small UI update on Docs
UI update for docs, plus fixes #24