WIP: Updates to the uikit

-[ ] Add a caption to the images