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