Update to include AssemblyBandit's color changes
Clean up AssemblyBandit's code