Added restart button
Added demo animations