Made some changes.

Like a boss