Minor updates on GUI