works on the bonus part, I think it works