Clean up draw.js