Cleaned up code

Signed-off-by: Cory Feitelson <cfeitelson@roovy.com>