Modify the layout slightly

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