code works now

Signed-off-by: Jason Kuster <unexpectedapocalypse@gmail.com>