Fixing unstaged lines (oops)