Functionality for both games added