Prepare code for small screens