update code

Signed-off-by: Caleb Cushing <xenoterracide@gmail.com>