Implemented suggestion by @leejjoon to use matplotlib transformations