Added restart iframe button

Cleaned code