fixed solution, skipped numpy