Yes We can: Watermarking machine learning models beyond classification