simple code fixes