Almost there! Fix more errors!

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