Increasing the Generalisaton Capacity of Conditional VAEs