Minor style change
Added comments
Added in a simple drift with likely incorrect math