Added better solution to the bonus.