Minor changes to get the app to compile