updated to work on smaller devices