Code cleanup & misc. fixes