Adding unittest.rb for testing the code
Implementing code_generation.rb module