added suggestions to code