small style changes
adding short code