Removed unneeded button and fixed package issue