refactor!

added the draw functions to their respective files