Layout fixes

Updated with a mobile layout