Improvements in the mock snippets

Added return action, once and repeatedly behaviours