trying to fix code to work in mobile section