Fixing minor errors regarding adding buttons