Library Works with UNO but not Mega 2560

I've been working on an arcade game using a UNO rev3 but I am starting to run out of storage for my resistive touchscreen LCD as well as I/Os. So I switched to the Mega 2560. However the touch sensing library will not upload to the Mega but it will to the UNO. I switch all the ports and board settings when I connect it so it is not the IDE. The error I get is:

Arduino: 1.6.11 (Mac OS X), Board: "Arduino/Genuino Mega or Mega 2560, ATmega2560 (Mega 2560)"
Arduino: 1.6.11 (Mac OS X), Board: "Arduino/Genuino Mega or Mega 2560, ATmega2560 (Mega 2560)"

In file included from /Users/admin/Documents/Arduino/libraries/TouchScreen/examples/touchscreendemo/touchscreendemo.pde:6:0:
/Users/admin/Documents/Arduino/libraries/TouchScreen/TouchScreen.h:51:12: error: 'RwReg' does not name a type
volatile RwReg *xp_port, *yp_port, *xm_port, *ym_port;
^
/Users/admin/Documents/Arduino/libraries/TouchScreen/TouchScreen.h:52:3: error: 'RwReg' does not name a type
RwReg xp_pin, xm_pin, yp_pin, ym_pin;
^
exit status 1
Error compiling for board Arduino/Genuino Mega or Mega 2560.

This report would have more information with
"Show verbose output during compilation"
option enabled in File -> Preferences.

I don't know enough about libraries to change the .h file to fix this. I was hoping someone could point me in the right direction on making it compatible with the Mega because this is the best/only library that works well for me. Thanks.

You can download the library here.

In TouchScreen.h change

#if defined(__AVR_ATmega328P__) || defined(__AVR_ATmega32U4__)

to

#if defined(__AVR_ATmega328P__) || defined(__AVR_ATmega32U4__) || defined(__AVR_ATmega2560__)

Awesome thanks so much! It works