Can probably improve this but it's ok for now.