Fixing code

Adding and fixing the code.