More clean-up

Signed-off-by: Alec Ari <neotheuser@ymail.com>