added restore button, fixed animations