Small Updates

Added code to into file.