Minor changes. Added missing library