initial changes

[adamgayoso]

I got the changes started. Did some cleanup and implemented the batch_removal function using some scvi-tools convention.

I think the predict function could really be cleaned up and easier to use.

I'm also not sure why the plotting functions are in the class.