Add back the su - around the oh-my-zsh install command.
Move the chown for vagrant's homedir to the end of the section.