Removed unneeded code. Added custom button capability.