Stylistcic fixes