Add loading state and fix minor UI alignment bugs

[kelvinkoon]

Please fill out the following before requesting review on this PR

Description

Testing Done

Resolved Issues

Length Justification

Review Checklist

It is the reviewers responsibility to also make sure every item here has been covered

• [ ] Function & Class comments: All function definitions should have comments, which are checked by the linters (ie. pylint).
• [ ] Remove all commented out code
• [ ] Remove extra print statements: for example, those just used for testing
• [ ] Resolve all TODO's: All TODO (or similar) statements should either be completed or associated with a github issue