Fine tuning + adding shutdown button