Made some changes.

Signed-off-by: Zach Bennett <xslingshot@gmail.com>