Moved draw function to another file to easily group functions