DOC: Add minor docstring to draw.js
TST: Fix tests in ui.js