Refactored buttons and init method.