Fix all the things

Signed-off-by: Rohan Mathur <rohanmathur34@gmail.com>