Cleaned up code

Signed-off-by: Alessandro D'Aquino (elara) <alex@midokura.com>