adding appropriate code blocks