Applied notes from review. Fixed formatting