fixed typo and added one more feature to look into