Added basic layout for draw function