WIP. Work on docs and button