Added else statment and test