some small adjustments to make it work better