Removed numpy library as it's no longer used