Added initial version of code