Not hard to do this, also not sure whether or not it can be improved.