Bug fixesssssssssssssssssszzzzzzzzzzzzzzzzzzzzzz