Applied some changes to have a margin on the side