* Initialising button before attempting to enable or disable it * Updating class reference * Adding exception use case * Removing unused use case * Including Exception class for classes where exception is used.