Improved UI

Signed-off-by: Gibran Gonzalez <haibrayn@hotmail.com>