Added restart button, drawing works fine now without weird bug