Adding styling fixes