minor edits

minor edits, still kinda pseudo code but just needed to get something
done in 5 mins.  DONT BE A BITCH.