Removed some imports, changed mutation class picker.