Fixes

Corrected buttons taken out of the last commit