Changed it to numpy version