Fixing code for the app.