Adds Button class, removes unneeded imports.