change code to support both 3.5 and 4 inch screen