[COS] Clean Code

Added protocols and Closures