Added code for testing purposes