Added support for multiple classes and method calls. Currently extremely limited. Lots of hardcoded assumptions that have to be changed (marked with //TODO)