Updated copyright and quick bug fix