Added a small batch of code