Added main element, added some style corrections for mobile